其他
一行代码使用雷达图比较不同模型指标的差异
构建多个不同的模型,选择最优的模型,今天来学习一个函数,使用这个函数快速比较不同模型的各项参数,并且一行代码可视化模型指标的差异。
1. 加载R包
加载需要用到的R包。
library(performance)
library(see)
library(survival)
library(tidyverse)
2. 加载数据
使用lung数据集中的数据进行演示,并简单进行数据处理。
dt <- lung %>%
na.omit() %>%
mutate(
status = ifelse(status == '1', 0, 1),
sex = factor(sex)
)
glimpse(dt)
3. 构建模型
m0 <- glm(status ~ age + sex + ph.ecog, data = dt, family = "binomial")
m1 <- glm(status ~ age + pat.karno + ph.ecog, data = dt, family = "binomial")
m2 <- glm(status ~ age + meal.cal + ph.ecog, data = dt, family = "binomial")
4. 比较模型
可以使用compare_performance()函数比较不同的模型。
compare_performance(m0, m1, m2)
如上,输出模型的质量评估指标不同数据。
5. 可视化模型指标差异
可以使用plot()函数可视化模型的不同指标,绘制雷达图。
plot(compare_performance(m0, m1, m2))
参考资料
performance包帮助文件。
《完》
关注下方公众号,分享更多更好玩的R语言知识。
觉得有帮助的请点赞、分享、在看走起!
点个在看,SCI马上发表。